package com.ruoyi.common.utils;

import org.apache.commons.lang3.time.DateFormatUtils;

import java.lang.management.ManagementFactory;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.time.*;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoUnit;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * 时间工具类
 *
 * @author ruoyi
 */
public class DateUtils extends org.apache.commons.lang3.time.DateUtils {
    public static String YYYY = "yyyy";

    public static String YYYY_MM = "yyyy-MM";

    public static String YYYY_MM_DD = "yyyy-MM-dd";

    public static String YYYYMMDDHHMMSS = "yyyyMMddHHmmss";

    public static String YYYY_MM_DD_HH_MM_SS = "yyyy-MM-dd HH:mm:ss";

    public static final DateTimeFormatter YYYY_MM_DD_FORMATTER = DateTimeFormatter.ofPattern(YYYY_MM_DD).withZone(ZoneId.systemDefault());

    public static final DateTimeFormatter YYYY_MM_FORMATTER = DateTimeFormatter.ofPattern(YYYY_MM).withZone(ZoneId.systemDefault());

    public static final DateTimeFormatter YYYY_MM_DD_HH_MM_SS_FORMATTER = DateTimeFormatter.ofPattern(YYYY_MM_DD_HH_MM_SS).withZone(ZoneId.systemDefault());

    private static String[] parsePatterns = {
            "yyyy-MM-dd", "yyyy-MM-dd HH:mm:ss", "yyyy-MM-dd HH:mm", "yyyy-MM",
            "yyyy/MM/dd", "yyyy/MM/dd HH:mm:ss", "yyyy/MM/dd HH:mm", "yyyy/MM",
            "yyyy.MM.dd", "yyyy.MM.dd HH:mm:ss", "yyyy.MM.dd HH:mm", "yyyy.MM"};

    /**
     * 获取当前Date型日期
     *
     * @return Date() 当前日期
     */
    public static Date getNowDate() {
        return new Date();
    }

    /**
     * 获取当前日期, 默认格式为yyyy-MM-dd
     *
     * @return String
     */
    public static String getDate() {
        return dateTimeNow(YYYY_MM_DD);
    }

    public static final String getTime() {
        return dateTimeNow(YYYY_MM_DD_HH_MM_SS);
    }

    public static final String dateTimeNow() {
        return dateTimeNow(YYYYMMDDHHMMSS);
    }

    public static final String dateTimeNow(final String format) {
        return parseDateToStr(format, new Date());
    }

    public static final String dateTime(final Date date) {
        return parseDateToStr(YYYY_MM_DD, date);
    }

    public static final String parseDateToStr(final String format, final Date date) {
        return new SimpleDateFormat(format).format(date);
    }

    public static final Date dateTime(final String format, final String ts) {
        try {
            return new SimpleDateFormat(format).parse(ts);
        } catch (ParseException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 日期路径 即年/月/日 如2018/08/08
     */
    public static final String datePath() {
        Date now = new Date();
        return DateFormatUtils.format(now, "yyyy/MM/dd");
    }

    /**
     * 日期路径 即年/月/日 如20180808
     */
    public static final String dateTime() {
        Date now = new Date();
        return DateFormatUtils.format(now, "yyyyMMdd");
    }

    /**
     * 日期型字符串转化为日期 格式
     */
    public static Date parseDate(Object str) {
        if (str == null) {
            return null;
        }
        try {
            return parseDate(str.toString(), parsePatterns);
        } catch (ParseException e) {
            return null;
        }
    }

    /**
     * 获取服务器启动时间
     */
    public static Date getServerStartDate() {
        long time = ManagementFactory.getRuntimeMXBean().getStartTime();
        return new Date(time);
    }

    /**
     * 计算相差天数
     */
    public static int differentDaysByMillisecond(Date date1, Date date2) {
        return Math.abs((int) ((date2.getTime() - date1.getTime()) / (1000 * 3600 * 24)));
    }

    /**
     * 计算时间差
     *
     * @param endDate   最后时间
     * @param startTime 开始时间
     * @return 时间差（天/小时/分钟）
     */
    public static String timeDistance(Date endDate, Date startTime) {
        long nd = 1000 * 24 * 60 * 60;
        long nh = 1000 * 60 * 60;
        long nm = 1000 * 60;
        // long ns = 1000;
        // 获得两个时间的毫秒时间差异
        long diff = endDate.getTime() - startTime.getTime();
        // 计算差多少天
        long day = diff / nd;
        // 计算差多少小时
        long hour = diff % nd / nh;
        // 计算差多少分钟
        long min = diff % nd % nh / nm;
        // 计算差多少秒//输出结果
        // long sec = diff % nd % nh % nm / ns;
        return day + "天" + hour + "小时" + min + "分钟";
    }

    /**
     * 增加 LocalDateTime ==> Date
     */
    public static Date toDate(LocalDateTime temporalAccessor) {
        ZonedDateTime zdt = temporalAccessor.atZone(ZoneId.systemDefault());
        return Date.from(zdt.toInstant());
    }

    /**
     * 增加 LocalDate ==> Date
     */
    public static Date toDate(LocalDate temporalAccessor) {
        LocalDateTime localDateTime = LocalDateTime.of(temporalAccessor, LocalTime.of(0, 0, 0));
        ZonedDateTime zdt = localDateTime.atZone(ZoneId.systemDefault());
        return Date.from(zdt.toInstant());
    }

    /**
     * 生成现在日期前day天到昨天的日期序列，yyyy-MM-dd格式
     *
     * @param day 现在日期前day天到昨天
     * @return 日期序列
     */
    public static List<String> generateDateList(int day) {
        List<String> dateList = new ArrayList<>(day);
        for (int i = 1; i <= day; i++) {
            LocalDate date = LocalDate.now().minusDays(i);
            dateList.add(date.toString());
        }
        return dateList;
    }

    /**
     * 生成指定年份的月份序列，yyyy-MM格式，如果是当年只返回到当前月份的序列，如当前为2024-07，只返回到7月的序列
     *
     * @param year 指定年份，不指定就为当年
     * @return 指定年份的月份序列
     */
    public static List<String> generateYearMonthList(Integer year) {
        YearMonth lastMonth = year == null ? YearMonth.now() : year == Year.now().getValue() ? YearMonth.now() : YearMonth.of(year, Month.DECEMBER);
        List<String> yearMonthList = new ArrayList<>();
        for (int i = 1; i <= lastMonth.getMonthValue(); i++) {
            yearMonthList.add(YearMonth.of(lastMonth.getYear(), i).toString());
        }
        return yearMonthList;
    }


    /**
     * 生成指定年月的日期序列，yyyy-MM-dd格式，如果是当月只返回到当前日的序列，如当前为2024-07-20，只返回到7月20日的序列
     *
     * @param yearMonth 指定年月，不指定就为当年月
     * @return 指定年月的日期序列
     */
    public static List<String> generateDateList(YearMonth yearMonth) {
        LocalDate lastDate;
        YearMonth thisYearMonth = YearMonth.now();
        if (yearMonth == null || thisYearMonth.equals(yearMonth)) {
            lastDate = LocalDate.now();
        } else {
            lastDate = LocalDate.of(yearMonth.getYear(), yearMonth.getMonth(), yearMonth.lengthOfMonth());
        }
        List<String> yearMonthList = new ArrayList<>();
        for (int i = 1; i <= lastDate.getDayOfMonth(); i++) {
            yearMonthList.add(LocalDate.of(lastDate.getYear(), lastDate.getMonth(), i).toString());
        }
        return yearMonthList;
    }

    /**
     * 生成指定两个日期之间的日期序列，包括指定的两个日期
     * <br>两个日期可以相同，相同时返回单个日期<code>list</code>
     * <br>指定的两个日期不分先后
     * <br>仅有一个日期不为<code>null</code>时返回该日期<code>list</code>
     * <br>两个日期均为<code>null</code>时返回{@link Collections#emptyList()}
     *
     * @param date1 日期1
     * @param date2 日期2
     * @return 正序的<code>LocalDate</code>日期序列
     */
    public static List<LocalDate> generateLocalDateListBetweenTwoDate(LocalDate date1, LocalDate date2) {
        if (date1 == null && date2 == null) {
            return Collections.emptyList();
        } else if (date1 == null) {
            return Collections.singletonList(date2);
        } else if (date2 == null) {
            return Collections.singletonList(date1);
        }
        LocalDate begin, end;
        if (date1.isAfter(date2)) {
            begin = date2;
            end = date1;
        } else {
            begin = date1;
            end = date2;
        }
        long numOfDaysBetween = ChronoUnit.DAYS.between(begin, end);
        return Stream.iterate(begin, date -> date.plusDays(1))
                .limit(numOfDaysBetween + 1)
                .collect(Collectors.toList());
    }

    /**
     * 生成指定两个日期之间的日期序列，包括指定的两个日期
     * <br>两个日期可以相同，相同时返回单个日期<code>list</code>
     * <br>指定的两个日期不分先后
     * <br>指定的两个日期字符串必须为<code>yyyy-MM-dd</code>格式
     * <br>仅有一个日期字符串合法时返回该日期<code>list</code>
     * <br>两个日期字符串均为<code>null</code>或<code>""</code>或<strong>非法日期格式</strong>则返回{@link Collections#emptyList()}
     *
     * @param date1 日期1
     * @param date2 日期2
     * @return 正序的<code>LocalDate</code>日期序列
     */
    public static List<LocalDate> generateLocalDateListBetweenTwoDate(String date1, String date2) {
        LocalDate d1 = null, d2 = null;
        try {
            d1 = LocalDate.parse(date1, YYYY_MM_DD_FORMATTER);
        } catch (Exception ignored) {
        }
        try {
            d2 = LocalDate.parse(date2, YYYY_MM_DD_FORMATTER);
        } catch (Exception ignored) {
        }
        return generateLocalDateListBetweenTwoDate(d1, d2);
    }

    /**
     * 生成指定两个日期之间的日期序列，包括指定的两个日期
     * <br>两个日期可以相同，相同时返回单个日期<code>list</code>
     * <br>指定的两个日期不分先后
     * <br>指定的两个日期字符串必须为<code>yyyy-MM-dd</code>格式
     * <br>仅有一个日期字符串合法时返回该日期<code>list</code>
     * <br>两个日期字符串均为<code>null</code>或<code>""</code>或<strong>非法日期格式</strong>则返回{@link Collections#emptyList()}
     *
     * @param date1 日期1
     * @param date2 日期2
     * @return 正序的日期序列字符串，<code>yyyy-MM-dd</code>格式
     */
    public static List<String> generateDateStringListBetweenTwoDate(String date1, String date2) {
        return generateLocalDateListBetweenTwoDate(date1, date2).stream().map(LocalDate::toString).collect(Collectors.toList());
    }

    /**
     * 生成指定两个年月之间的年月序列，包括指定的两个年月
     * <br>两个年月可以相同，相同时返回单个年月<code>list</code>
     * <br>指定的两个年月不分先后
     * <br>仅有一个年月不为<code>null</code>时返回该年月<code>list</code>
     * <br>两个年月均为<code>null</code>时返回{@link Collections#emptyList()}
     *
     * @param yearMonth1 年月1
     * @param yearMonth2 年月2
     * @return 正序的<code>YearMonth</code>年月序列
     */
    public static List<YearMonth> generateYearMonthListBetweenTwoYearMonth(YearMonth yearMonth1, YearMonth yearMonth2) {
        if (yearMonth1 == null && yearMonth2 == null) {
            return Collections.emptyList();
        } else if (yearMonth1 == null) {
            return Collections.singletonList(yearMonth2);
        } else if (yearMonth2 == null) {
            return Collections.singletonList(yearMonth1);
        }
        YearMonth begin, end;
        if (yearMonth1.isAfter(yearMonth2)) {
            begin = yearMonth2;
            end = yearMonth1;
        } else {
            begin = yearMonth1;
            end = yearMonth2;
        }
        long numOfMonthsBetween = ChronoUnit.MONTHS.between(begin, end);
        return Stream.iterate(begin, yearMonth -> yearMonth.plusMonths(1))
                .limit(numOfMonthsBetween + 1)
                .collect(Collectors.toList());
    }

    /**
     * 生成指定两个年月之间的年月序列，包括指定的两个年月
     * <br>两个年月可以相同，相同时返回单个年月<code>list</code>
     * <br>指定的两个年月不分先后
     * <br>指定的两个年月字符串必须为<code>yyyy-MM</code>格式
     * <br>仅有一个年月字符串合法时返回该年月<code>list</code>
     * <br>两个年月字符串均为<code>null</code>或<code>""</code>或<strong>非法年月格式</strong>则返回{@link Collections#emptyList()}
     *
     * @param yearMonth1 年月1
     * @param yearMonth2 年月2
     * @return 正序的<code>YearMonth</code>年月序列
     */
    public static List<YearMonth> generateYearMonthListBetweenTwoYearMonth(String yearMonth1, String yearMonth2) {
        YearMonth d1 = null, d2 = null;
        try {
            d1 = YearMonth.parse(yearMonth1, YYYY_MM_FORMATTER);
        } catch (Exception ignored) {
        }
        try {
            d2 = YearMonth.parse(yearMonth2, YYYY_MM_FORMATTER);
        } catch (Exception ignored) {
        }
        return generateYearMonthListBetweenTwoYearMonth(d1, d2);
    }

    /**
     * 生成指定两个年月之间的年月序列，包括指定的两个年月
     * <br>两个年月可以相同，相同时返回单个年月<code>list</code>
     * <br>指定的两个年月不分先后
     * <br>指定的两个年月字符串必须为<code>yyyy-MM</code>格式
     * <br>仅有一个年月字符串合法时返回该年月<code>list</code>
     * <br>两个年月字符串均为<code>null</code>或<code>""</code>或<strong>非法年月格式</strong>则返回{@link Collections#emptyList()}
     *
     * @param yearMonth1 年月1
     * @param yearMonth2 年月2
     * @return 正序的年月序列字符串，<code>yyyy-MM</code>格式
     */
    public static List<String> generateYearMonthStringListBetweenTwoYearMonth(String yearMonth1, String yearMonth2) {
        return generateYearMonthListBetweenTwoYearMonth(yearMonth1, yearMonth2).stream().map(YearMonth::toString).collect(Collectors.toList());
    }
}
